package org.fyv233.missingno_craft.datagen;

import net.fabricmc.fabric.api.datagen.v1.FabricDataOutput;
import net.fabricmc.fabric.api.datagen.v1.provider.FabricTagProvider;
import net.minecraft.registry.RegistryWrapper;
import net.minecraft.registry.tag.ItemTags;
import org.fyv233.missingno_craft.item.ModItems;

import java.util.concurrent.CompletableFuture;

public class ModItemsTagsProvider extends FabricTagProvider.ItemTagProvider{

    public ModItemsTagsProvider(FabricDataOutput output, CompletableFuture<RegistryWrapper.WrapperLookup> registriesFuture) {
        super(output, registriesFuture);
    }

    @Override
    protected void configure(RegistryWrapper.WrapperLookup wrapperLookup) {
        getOrCreateTagBuilder(ItemTags.MINING_ENCHANTABLE)
                .add(ModItems.MISSINGNO_PICKAXE)
                .add(ModItems.MISSINGNO_HOE)
                .add(ModItems.MISSINGNO_AXE);

        getOrCreateTagBuilder(ItemTags.SWORD_ENCHANTABLE)
                .add(ModItems.MISSINGNO_SWORD);

        getOrCreateTagBuilder(ItemTags.TRIMMABLE_ARMOR)
                .add(ModItems.MISSINGNO_HELMET)
                .add(ModItems.MISSINGNO_CHESTPLATE)
                .add(ModItems.MISSINGNO_BOOTS)
                .add(ModItems.MISSINGNO_LEGGINGS);
    }
}
